import argparse


def my_args():
    # 这个接口能通过命令行传参, 且可以设定默认值
    parser = argparse.ArgumentParser()
    # 名称, 描述, 类型, 默认值
    parser.add_argument('-device', '--device',
                        type=str, default='cpu')
    parser.add_argument('-Epoch', '--Epoch',
                        type=int, default=100)
    parser.add_argument('-learning_rate', '--learning_rate',
                        type=float, default=1e-2)
    parser.add_argument('-step_size', '--step_size',
                        type=int, default=5)
    parser.add_argument('-gamma', '--gamma',
                        type=float, default=0.1)
    return parser.parse_args()
